Node / ReticulumProjects / MeshChatX.git / files / tests / backend / test_community_interfaces.py

Displaying Raw • Download

Tff7b72from T7ee787unittestT7ee787.T7ee787mock Tff7b72import Te6edf3MagicMockTb4b4b4, Te6edf3patch

Tff7b72import T7ee787pytest

Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787community_interfaces Tff7b72import Te6edf3CommunityInterfacesManager

Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787rnstatus_handler Tff7b72import Te6edf3RNStatusHandler

Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio

Tff7b72async Tff7b72def Td2a8fftest_community_interfaces_manager_health_checkTb4b4b4(Tb4b4b4)Tb4b4b4:

Te6edf3manager Tff7b72= Te6edf3CommunityInterfacesManagerTb4b4b4(Tb4b4b4)

T8b949e# Mock check_health to return True for first, False for second

Tff7b72with Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(

Te6edf3CommunityInterfacesManagerTb4b4b4,

Ta5d6ff"Ta5d6ffcheck_healthTa5d6ff"Tb4b4b4,

Te6edf3side_effectTff7b72=Tb4b4b4[Tff7b72TrueTb4b4b4, Tff7b72FalseTb4b4b4]Tb4b4b4,

Tb4b4b4)Tb4b4b4:

Te6edf3interfaces Tff7b72= Tff7b72await Te6edf3managerTff7b72.Td2a8ffget_interfacesTb4b4b4(Tb4b4b4)

Tff7b72assert Tffa657lenTb4b4b4(Te6edf3interfacesTb4b4b4) Tff7b72== T79c0ff2

T8b949e# First one should be online because we sort by online status

Tff7b72assert Te6edf3interfacesTb4b4b4[T79c0ff0Tb4b4b4]Tb4b4b4[Ta5d6ff"Ta5d6ffonlineTa5d6ff"Tb4b4b4] Tff7b72is Tff7b72True

Tff7b72assert Te6edf3interfacesTb4b4b4[T79c0ff1Tb4b4b4]Tb4b4b4[Ta5d6ff"Ta5d6ffonlineTa5d6ff"Tb4b4b4] Tff7b72is Tff7b72False

T8b949e# Check that we have both online and offline

Te6edf3online_count Tff7b72= Tffa657sumTb4b4b4(T79c0ff1 Tff7b72for Te6edf3iface Tff7b72in Te6edf3interfaces Tff7b72if Te6edf3ifaceTb4b4b4[Ta5d6ff"Ta5d6ffonlineTa5d6ff"Tb4b4b4]Tb4b4b4)

Tff7b72assert Te6edf3online_count Tff7b72== T79c0ff1

Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio

Tff7b72async Tff7b72def Td2a8fftest_rnstatus_integration_simulatedTb4b4b4(Tb4b4b4)Tb4b4b4:

T8b949e# Simulate how rnstatus would see these interfaces if they were added

Te6edf3mock_reticulum Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3mock_reticulumTff7b72.Td2a8ffget_interface_statsTff7b72.Td2a8ffreturn_value Tff7b72= Tb4b4b4{

Ta5d6ff"Ta5d6ffinterfacesTa5d6ff"Tb4b4b4: Tb4b4b4[

Tb4b4b4{

Ta5d6ff"Ta5d6ffnameTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6ffnoDNS1Ta5d6ff"Tb4b4b4,

Ta5d6ff"Ta5d6ffstatusTa5d6ff"Tb4b4b4: Tff7b72TrueTb4b4b4,

Ta5d6ff"Ta5d6ffrxbTa5d6ff"Tb4b4b4: T79c0ff100Tb4b4b4,

Ta5d6ff"Ta5d6fftxbTa5d6ff"Tb4b4b4: T79c0ff200Tb4b4b4,

Tb4b4b4}Tb4b4b4,

Tb4b4b4{

Ta5d6ff"Ta5d6ffnameTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6ffQuad4 TCP Node 1Ta5d6ff"Tb4b4b4,

Ta5d6ff"Ta5d6ffstatusTa5d6ff"Tb4b4b4: Tff7b72FalseTb4b4b4,

Ta5d6ff"Ta5d6ffrxbTa5d6ff"Tb4b4b4: T79c0ff0Tb4b4b4,

Ta5d6ff"Ta5d6fftxbTa5d6ff"Tb4b4b4: T79c0ff0Tb4b4b4,

Tb4b4b4}Tb4b4b4,

Tb4b4b4]Tb4b4b4,

Tb4b4b4}

Te6edf3handler Tff7b72= Te6edf3RNStatusHandlerTb4b4b4(Te6edf3mock_reticulumTb4b4b4)

Te6edf3status Tff7b72= Te6edf3handlerTff7b72.Td2a8ffget_statusTb4b4b4(Tb4b4b4)

Tff7b72assert Tffa657lenTb4b4b4(Te6edf3statusTb4b4b4[Ta5d6ff"Ta5d6ffinterfacesTa5d6ff"Tb4b4b4]Tb4b4b4) Tff7b72== T79c0ff2

Tff7b72assert Te6edf3statusTb4b4b4[Ta5d6ff"Ta5d6ffinterfacesTa5d6ff"Tb4b4b4]Tb4b4b4[T79c0ff0Tb4b4b4]Tb4b4b4[Ta5d6ff"Ta5d6ffnameTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffnoDNS1Ta5d6ff"

Tff7b72assert Te6edf3statusTb4b4b4[Ta5d6ff"Ta5d6ffinterfacesTa5d6ff"Tb4b4b4]Tb4b4b4[T79c0ff0Tb4b4b4]Tb4b4b4[Ta5d6ff"Ta5d6ffstatusTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffUpTa5d6ff"

Tff7b72assert Te6edf3statusTb4b4b4[Ta5d6ff"Ta5d6ffinterfacesTa5d6ff"Tb4b4b4]Tb4b4b4[T79c0ff1Tb4b4b4]Tb4b4b4[Ta5d6ff"Ta5d6ffnameTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffQuad4 TCP Node 1Ta5d6ff"

Tff7b72assert Te6edf3statusTb4b4b4[Ta5d6ff"Ta5d6ffinterfacesTa5d6ff"Tb4b4b4]Tb4b4b4[T79c0ff1Tb4b4b4]Tb4b4b4[Ta5d6ff"Ta5d6ffstatusTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffDownTa5d6ff"

Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio

Tff7b72async Tff7b72def Td2a8fftest_community_interfaces_dynamic_updateTb4b4b4(Tb4b4b4)Tb4b4b4:

Te6edf3manager Tff7b72= Te6edf3CommunityInterfacesManagerTb4b4b4(Tb4b4b4)

T8b949e# Mock check_health to return different values over time

Tff7b72with Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3CommunityInterfacesManagerTb4b4b4, Ta5d6ff"Ta5d6ffcheck_healthTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_checkTb4b4b4:

T8b949e# First check: all online

Te6edf3mock_checkTff7b72.Td2a8ffreturn_value Tff7b72= Tff7b72True

Te6edf3ifaces1 Tff7b72= Tff7b72await Te6edf3managerTff7b72.Td2a8ffget_interfacesTb4b4b4(Tb4b4b4)

Tff7b72assert Tffa657allTb4b4b4(Te6edf3ifaceTb4b4b4[Ta5d6ff"Ta5d6ffonlineTa5d6ff"Tb4b4b4] Tff7b72for Te6edf3iface Tff7b72in Te6edf3ifaces1Tb4b4b4)

T8b949e# Force update by clearing last_check and mock all offline

Te6edf3managerTff7b72.Td2a8fflast_check Tff7b72= T79c0ff0

Te6edf3mock_checkTff7b72.Td2a8ffreturn_value Tff7b72= Tff7b72False

Te6edf3ifaces2 Tff7b72= Tff7b72await Te6edf3managerTff7b72.Td2a8ffget_interfacesTb4b4b4(Tb4b4b4)

Tff7b72assert Tffa657allTb4b4b4(Tff7b72not Te6edf3ifaceTb4b4b4[Ta5d6ff"Ta5d6ffonlineTa5d6ff"Tb4b4b4] Tff7b72for Te6edf3iface Tff7b72in Te6edf3ifaces2Tb4b4b4)